package com.sheng.leetcode.year2023.month01.day31;

import org.junit.Test;

/**
 * @author liusheng
 * @date 2023/01/31
 * <p>
 * 2319. 判断矩阵是否是一个 X 矩阵<p>
 * <p>
 * 如果一个正方形矩阵满足下述 全部 条件，则称之为一个 X 矩阵 ：<p>
 * 矩阵对角线上的所有元素都 不是 0<p>
 * 矩阵中所有其他元素都是 0<p>
 * 给你一个大小为 n x n 的二维整数数组 grid ，表示一个正方形矩阵。如果 grid 是一个 X 矩阵 ，返回 true ；否则，返回 false 。<p>
 * <p>
 * 示例 1：<p>
 * 输入：grid = [[2,0,0,1],[0,3,1,0],[0,5,2,0],[4,0,0,2]]<p>
 * 输出：true<p>
 * 解释：矩阵如上图所示。<p>
 * X 矩阵应该满足：绿色元素（对角线上）都不是 0 ，红色元素都是 0 。<p>
 * 因此，grid 是一个 X 矩阵。<p>
 * <p>
 * 示例 2：<p>
 * 输入：grid = [[5,7,0],[0,3,1],[0,5,0]]<p>
 * 输出：false<p>
 * 解释：矩阵如上图所示。<p>
 * X 矩阵应该满足：绿色元素（对角线上）都不是 0 ，红色元素都是 0 。<p>
 * 因此，grid 不是一个 X 矩阵。<p>
 * <p>
 * 提示：<p>
 * n == grid.length == grid[i].length<p>
 * 3 <= n <= 100<p>
 * 0 <= grid[i][j] <= 10^5<p>
 * <p>
 * 来源：力扣（LeetCode）<p>
 * 链接：<a href="https://leetcode.cn/problems/check-if-matrix-is-x-matrix">2319. 判断矩阵是否是一个 X 矩阵</a><p>
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。<p>
 */
public class LeetCode2319 {

    @Test
    public void test01() {
        int[][] grid = {{2, 0, 0, 1}, {0, 3, 1, 0}, {0, 5, 2, 0}, {4, 0, 0, 2}};
//        int[][] grid = {{5, 7, 0}, {0, 3, 1}, {0, 5, 0}};
        System.out.println(new Solution().checkXMatrix(grid));
    }
}

class Solution {
    public boolean checkXMatrix(int[][] grid) {
        int n = grid.length;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (i == j || i + j == n - 1) {
                    if (grid[i][j] == 0) {
                        return false;
                    }
                } else if (grid[i][j] != 0) {
                    return false;
                }
            }
        }
        return true;
    }
}
